Split() function in PySpark